package com.xkcyy.one.mapper;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.xkcyy.one.entity.AgentTool;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Select;

import java.util.List;

/**
 * 智能体工具关联数据访问层
 * 
 * @author yuand
 */
@Mapper
public interface AgentToolMapper extends BaseMapper<AgentTool> {

    /**
     * 根据智能体ID查询关联的工具ID列表
     *
     * @param agentId 智能体ID
     * @return 工具ID列表
     */
    @Select("SELECT tool_id FROM agent_tool WHERE agent_id = #{agentId}")
    List<Long> selectToolIdsByAgentId(Long agentId);

    /**
     * 根据工具ID查询关联的智能体ID列表
     *
     * @param toolId 工具ID
     * @return 智能体ID列表
     */
    @Select("SELECT agent_id FROM agent_tool WHERE tool_id = #{toolId}")
    List<Long> selectAgentIdsByToolId(Long toolId);
}